/*
 * Copyright (C) 2013 EfficiOS Inc.
 *
 * SPDX-License-Identifier: GPL-2.0-only
 *
 */

#include <common/utils.hpp>

#include <stdio.h>
#include <string.h>
#include <tap/tap.h>

/* For error.h */
int lttng_opt_quiet = 1;
int lttng_opt_verbose = 3;
int lttng_opt_mi;

namespace {
struct valid_test_input {
	const char *input;
	uint64_t expected_result;
};

/* Valid test cases */
struct valid_test_input valid_tests_inputs[] = {
	{ "0", 0 },
	{ "1234", 1234 },
	{ "0x400", 1024 },
	{ "0300", 192 },
	{ "16k", 16384 },
	{ "128K", 131072 },
	{ "0x1234k", 4771840 },
	{ "32M", 33554432 },
	{ "1024G", 1099511627776ULL },
	{ "0X400", 1024 },
	{ "0x40a", 1034 },
	{ "0X40b", 1035 },
	{ "0x40C", 1036 },
	{ "0X40D", 1037 },
	{ "0x40e", 1038 },
	{ "0X40f", 1039 },
	{ "00", 0 },
	{ "0k", 0 },
	{ "0K", 0 },
	{ "0M", 0 },
	{ "0G", 0 },
	{ "00k", 0 },
	{ "00K", 0 },
	{ "00M", 0 },
	{ "00G", 0 },
	{ "0x0", 0 },
	{ "0X0", 0 },
	{ "0x0k", 0 },
	{ "0X0K", 0 },
	{ "0x0M", 0 },
	{ "0X0G", 0 },
	{ "0X40G", 68719476736ULL },
	{ "0300k", 196608 },
	{ "0300K", 196608 },
	{ "030M", 25165824 },
	{ "020G", 17179869184ULL },
	{ "0xa0k", 163840 },
	{ "0xa0K", 163840 },
	{ "0XA0M", 167772160 },
	{ "0xA0G", 171798691840ULL },
};
const int num_valid_tests = sizeof(valid_tests_inputs) / sizeof(valid_tests_inputs[0]);

/* Invalid test cases */
const char *invalid_tests_inputs[] = {
	"",	 " ",	  "-1",	 "k",	"4611686018427387904G",
	"0x40g", "08",	  "09",	 "0x",	"x0",
	"0xx0",	 "07kK",  "0xk", "0XM", "0xG",
	"0x0MM", "0X0GG", "0a",	 "0B",
};

const int num_invalid_tests = sizeof(invalid_tests_inputs) / sizeof(invalid_tests_inputs[0]);
} /* namespace */

static void test_utils_parse_size_suffix()
{
	uint64_t result;
	int ret;
	int i;

	/* Test valid cases */
	for (i = 0; i < num_valid_tests; i++) {
		char name[100];
		sprintf(name, "valid test case: %s", valid_tests_inputs[i].input);

		ret = utils_parse_size_suffix(valid_tests_inputs[i].input, &result);
		ok(ret == 0 && result == valid_tests_inputs[i].expected_result, "%s", name);
	}

	/* Test invalid cases */
	for (i = 0; i < num_invalid_tests; i++) {
		char name[100];
		sprintf(name, "invalid test case: %s", invalid_tests_inputs[i]);

		ret = utils_parse_size_suffix(invalid_tests_inputs[i], &result);
		ok(ret != 0, "%s", name);
	}
}

int main()
{
	plan_tests(num_valid_tests + num_invalid_tests);

	diag("utils_parse_size_suffix tests");

	test_utils_parse_size_suffix();

	return exit_status();
}
